from PyQt6.QtWidgets import (QMainWindow, QWidget, QVBoxLayout, 
                               QHBoxLayout, QPushButton, QStackedWidget)
from PyQt6.QtCore import Qt
from .components.task_config import TaskConfigWidget
from .components.menu import MenuWidget

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("AI创影")
        self.setMinimumSize(1200, 800)
        
        # 创建主窗口部件
        main_widget = QWidget()
        self.setCentralWidget(main_widget)
        
        # 创建水平布局
        layout = QHBoxLayout()
        main_widget.setLayout(layout)
        
        # 创建左侧菜单
        self.menu_widget = MenuWidget(self)
        layout.addWidget(self.menu_widget)
        
        # 创建右侧内容区域
        self.content_stack = QStackedWidget()
        layout.addWidget(self.content_stack)
        
        # 设置布局比例
        layout.setStretch(0, 1)  # 菜单占比
        layout.setStretch(1, 4)  # 内容区域占比
        
        # 初始化页面
        self.init_pages()
        
        # 设置样式
        self.setStyleSheet("""
            QMainWindow {
                background-color: #f5f5f5;
            }
        """)
    
    def init_pages(self):
        """初始化所有页面"""
        # 添加任务配置页面
        self.task_config = TaskConfigWidget()
        self.content_stack.addWidget(self.task_config)
        
    def switch_page(self, index):
        """切换页面"""
        self.content_stack.setCurrentIndex(index) 